package leetcode._07_贪心;

import org.junit.Test;

import java.util.Comparator;
import java.util.HashMap;
import java.util.PriorityQueue;

/**
 * @author pppppp
 * @date 2022/3/21 11:32
 * 在柠檬水摊上，每一杯柠檬水的售价为 5 美元。顾客排队购买你的产品，（按账单 bills 支付的顺序）一次购买一杯。
 * 每位顾客只买一杯柠檬水，然后向你付 5 美元、10 美元或 20 美元。你必须给每个顾客正确找零，也就是说净交易是每位顾客向你支付 5 美元。
 * 注意，一开始你手头没有任何零钱。
 * <p>
 * 给你一个整数数组 bills ，其中 bills[i] 是第 i 位顾客付的账。如果你能给每位顾客正确找零，返回 true ，否则返回 false 。
 * 示例 1：
 * 输入：bills = [5,5,5,10,20]
 * 输出：true
 * 解释：
 * 前 3 位顾客那里，我们按顺序收取 3 张 5 美元的钞票。
 * 第 4 位顾客那里，我们收取一张 10 美元的钞票，并返还 5 美元。
 * 第 5 位顾客那里，我们找还一张 10 美元的钞票和一张 5 美元的钞票。
 * 由于所有客户都得到了正确的找零，所以我们输出 true。
 * <p>
 * 示例 2：
 * 输入：bills = [5,5,10,10,20]
 * 输出：false
 * 解释：
 * 前 2 位顾客那里，我们按顺序收取 2 张 5 美元的钞票。
 * 对于接下来的 2 位顾客，我们收取一张 10 美元的钞票，然后返还 5 美元。
 * 对于最后一位顾客，我们无法退回 15 美元，因为我们现在只有两张 10 美元的钞票。
 * 由于不是每位顾客都得到了正确的找零，所以答案是 false。
 * 提示：
 * <p>
 * 1 <= bills.length <= 105
 * bills[i] 不是 5 就是 10 或是 20
 */
public class _860_柠檬水找零 {
    @Test
    public void T_0() {
        int[][] nums = {{5,5,10,20,5,5,5,5,5,5,5,5,5,10,5,5,20,5,20,5},{5, 5, 5, 10, 20}, {5, 5, 10, 10, 20}};
        boolean[] ans = {true,true, false};
        for (int i = 0; i < nums.length; i++) {
            System.out.println(lemonadeChange(nums[i]) == ans[i]);
        }
    }

    public boolean lemonadeChange(int[] bills) {
        if (bills[0] > 5) {
            return false;
        }
        int[] res = new int[3];
        res[0] = 1;
        for (int i = 1; i < bills.length; i++) {
            if (bills[i] == 5) {
                res[0]++;
            } else if (bills[i] == 10) {
                res[0]--;
                res[1]++;
            } else {
                if(res[1] > 0){
                    res[1]--;
                    res[0]--;
                }else {
                    res[0] -= 3;
                }
                res[2]++;
            }
            if (res[0] < 0 || res[1] < 0) {
                return false;
            }
        }
        return true;
    }
}
